package com.sean.dao;

import com.sean.User;
import com.sean.util.DbUtilHikari;
import cn.hutool.crypto.SecureUtil;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * 对用户表的操作
 * @author Wen Xiaoyu
 */
public class UserDao {
    /**
     * 用户的注册
     * @param uname 用户名
     * @param upsw 密码
     * @return 注册成功返回1，否则返回0
     */
    public int insertUser(String uname,String upsw)  {
        //1、声明一下变量备用
        //代表数据库连接
        Connection conn = null;
        //预编译对象，用pstmt执行sql语句
        PreparedStatement pstmt =null;
        //注册成功与否的的标志，如果是0注册失败，如果是1注册成功
        int flag=0;

        //2、获取数据库连接
        conn= DbUtilHikari.getConnection();
        //3、定义并准备sql语句
        String sql = "INSERT INTO users(username,userpsw) VALUES(?,?)";
        try {
            pstmt = conn.prepareStatement(sql);
            //4、给问号赋值
            pstmt.setString(1,uname);
            pstmt.setString(2,SecureUtil.md5(upsw));
            //5、执行sql语句（）下达执行指令
            flag = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }finally {
            //6、关闭数据库资源
            DbUtilHikari.close(pstmt,conn);
        }
        return flag;
    }

    /**
     * 登录操作
     * @param uname 用户名
     * @param upsw 密码
     * @return
     */
    public boolean login(String uname,String upsw){
        //flag是登录成功与否的标志。
        boolean flag = false;
        //conn表示数据库连接
        Connection conn = null;
        //pstmt是执行sql语句的
        PreparedStatement pstmt = null;
        //rs代表结果集，从数据库里查到的结果方到这里
        ResultSet rs  =null;


        //获得数据连接,从连接池里获取，不是直接从数据库打开连接
        conn = DbUtilHikari.getConnection();
        //定义并准备sql语句,登录是查询语句
        String sql = "SELECT * FROM users WHERE username=? AND userpsw=?";
        try {
            pstmt = conn.prepareStatement(sql);
            //给？赋值
            pstmt.setString(1,uname);
            pstmt.setString(2,SecureUtil.md5(upsw));
            //下达执行指令
            rs = pstmt.executeQuery();
            //判断rs结果集里有没有数据
            //如果next（）返回true，说明rs有数据
            if(rs.next()){
                flag = true;
            }else {
                flag = false;
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            DbUtilHikari.close(rs,pstmt,conn);
        }
        return flag;
    }

    /**
     * 按用户编号删除用户
     * @param userid 用户编号
     * @return 删除成功返回1，否则返回0
     */
    public int deleteUser(int userid){
        int flag = 0;
        //声明一些变量
        Connection conn = null;
        PreparedStatement pstmt = null;
        //获取数据库连接
        conn = DbUtilHikari.getConnection();
        //定义并准备sql语句
        String sql = "DELETE FROM users WHERE userid=?" ;
        try {
            pstmt = conn.prepareStatement(sql);
            pstmt.setInt(1,userid);
            //执行sql语句
            flag = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            //关闭资源（释放资源）
            DbUtilHikari.close(pstmt,conn);
        }

        return flag;

    }

    /**
     * 修改用户信息
     * @param userid 用户编号
     * @param uname 用户名字
     * @param userpsw 密码
     * @return 成功返回1，否则0
     */
    public int updateUser(int userid,String uname,String userpsw){
        int flag = 0;
        Connection conn = null;
        PreparedStatement pstmt = null;
        conn = DbUtilHikari.getConnection();
        String sql = "UPDATE users SET username=?,userpsw=? WHERE userid=?";
        try {
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1,uname);
            pstmt.setString(2,SecureUtil.md5(userpsw));
            pstmt.setInt(3,userid);
            flag = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            DbUtilHikari.close(pstmt,conn);
        }
        return flag;
    }

    /**
     * 查询所有的用户
     * @return
     */
    public List<User> selectAllUser(){
        //创建一个空集合，准备放一个用户对象
        List<User> userList = new ArrayList<User>();
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        conn = DbUtilHikari.getConnection();
        String sql = "SELECT userid,username,userpsw FROM users";

        try {
            pstmt = conn.prepareStatement(sql);
            rs = pstmt.executeQuery();
            //把rs里的数据转移到userList集合里
            while(rs.next()){
                //User当前是空的
                User user = new User();
                user.setUserid(rs.getInt("userid"));
                user.setUsername(rs.getString("username"));
                user.setUserpsw(rs.getString("userpsw"));
                userList.add(user);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            DbUtilHikari.close(rs,pstmt,conn);
        }

        return userList;

    }
}